hackerrank ruby enumerable collect problem